CUDA_VISIBLE_DEVICES=3 python UHD-Net/src/train.py \
  --train-dir ../../data/UHD-LL/training_set/\
  --valid-dir ../../data/UHD-LL/testing_set/ \
  --ckpt-save-path UHD-Net/ckpts_UHD_7/ \
  --nb-epochs 1000 \
  --batch-size 1\
  --loss l1 \
  --noise-type gaussian \
  --noise-param 25 \
  --crop_size 512 \
  --plot-stats \
  --cuda   
  # --ckpt_load_path
  #--crop-size 240 \      
  # --model swin \                                      
  
  CUDA_VISIBLE_DEVICES=0,1,2,3 python UHD-Net/src/train.py \
  --train-dir ../../data/UHD-LL/training_set/\
  --valid-dir ../../data/UHD-LL/testing_set/ \
  --ckpt-save-path UHD-Net/ckpts_UHD_7/ \
  --nb-epochs 1000 \
  --batch-size 2\
  --loss l1 \
  --noise-type gaussian \
  --noise-param 25 \
  --crop-size 30 \
  --plot-stats \
  --cuda   

#原版
  CUDA_VISIBLE_DEVICES=4,5 python UHD-Net/src/train.py \
  --train-dir ../../data/UHD-LL/training_set/\
  --valid-dir ../../data/UHD-LL/testing_set/ \
  --ckpt-save-path UHD-Net/ckpts_UHD_7/ \
  --nb-epochs 1000 \
  --batch-size 2\
  --loss l1 \
  --noise-type gaussian \
  --noise-param 25 \
  --crop-size 30 \
  --plot-stats \
  --cuda   